package com.gpt.article.service.impl.taskHandle.task;

import com.gpt.article.domain.GptUseToken;
import com.gpt.article.service.IGptTopicTitleService;
import com.gpt.article.service.IGptUseTokenService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.math.BigInteger;

@Service
public class OptUseTokenDb {

    private static Logger logger = LoggerFactory.getLogger(OptUseTokenDb.class);

    private static IGptUseTokenService gptUseTokenService;

    @Autowired
    public void setIGptUseTokenService(IGptUseTokenService gptUseTokenService) {
        OptUseTokenDb.gptUseTokenService = gptUseTokenService;
    }


    /**
     * 插入token花费信息
     * @param useToken
     * @return
     */
    public Long insertTokenInfo(GptUseToken useToken){
        GptUseToken gptUseToken = new GptUseToken();
        gptUseToken.setModuleName(useToken.getModuleName());
        gptUseToken.setBusinessId(useToken.getBusinessId());
        gptUseToken.setTaskId(useToken.getTaskId());
        gptUseToken.setUseDate(useToken.getUseDate());
        gptUseToken.setUseModel(useToken.getUseModel());
        gptUseToken.setPayMoney(useToken.getPayMoney());
        gptUseToken.setTotalTokens(useToken.getTotalTokens());
        gptUseToken.setPromptTokens(useToken.getPromptTokens());
        gptUseToken.setCompletionTokens(useToken.getCompletionTokens());
        gptUseToken.setElapsedTime(useToken.getElapsedTime());
        gptUseToken.setCreateBy(useToken.getCreateBy());
        Long fileId = gptUseTokenService.insertGptUseToken(gptUseToken);

        return fileId;
    }

    /**
     * 更新
     * @param useToken
     * @return
     */
    public boolean updateTolenInfo(GptUseToken useToken){
        gptUseTokenService.updateGptUseToken(useToken);
        return true;
    }

}
